package edu.biwu.request;

import jakarta.servlet.*;
import jakarta.servlet.http.*;
import jakarta.servlet.annotation.*;

import java.io.File;
import java.io.IOException;
import java.util.Collection;
import java.util.UUID;
import java.util.stream.Collectors;

//处理多文件上传
@WebServlet("/req/upload02")
@MultipartConfig
public class ReqUploadDemo02 extends HttpServlet {
    @Override
    protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        //1.利用request对象中的getParts()方法获取所有的上传文件信息
        //每个上传文件的信息对应一个Part,然后将所有的Part存储到一个集合中
        Collection<Part> uploadFiles = request.getParts();

        //2.遍历所有的Part(所有的上传文件)
        //例如上传: 1.jpg 2.txt 3.pdf
        for (Part uploadFile : uploadFiles) {
            String fileName = uploadFile.getSubmittedFileName();

            String extension = fileName.substring(fileName.lastIndexOf("."));// 截取文件的后缀名,例如: 1.jpg=>.jpg
            String newFileName = UUID.randomUUID() + extension;//dfdffd-12323-sddf.jpg

            //3.先选择一个服务端的存储位置
            File dir = new File("D:/uploadFile");
            if (!dir.exists()) {//如果指定的存储目录不存在,直接创建该目录
                dir.mkdirs();
            }

            //4.将上传的文件写入到服务端的某个位置,存储上传文件
            uploadFile.write(dir + "/" + newFileName);//直接将上传的文件写入指定位置
        }

        //5.上传成功后,响应上传成功提示
        String fileNames = uploadFiles.stream().map(uploadFile -> uploadFile.getSubmittedFileName())
                .collect(Collectors.joining(","));

        response.setContentType("text/html;charset=UTF-8");//防止响应中文乱码
        response.getWriter().write(fileNames+" 上传成功");
    }

    @Override
    protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        doGet(request, response);
    }
}
